#!/bin/bash

echo "================================"
echo "    数据预处理系统启动脚本"
echo "================================"
echo

# 检查Python是否安装
if ! command -v python3 &> /dev/null; then
    echo "错误: 未检测到Python3，请先安装Python 3.8+"
    exit 1
fi

# 显示Python版本
echo "Python版本: $(python3 --version)"

# 检查虚拟环境
if [[ "$VIRTUAL_ENV" == "" ]]; then
    echo "建议使用虚拟环境运行此项目"
    read -p "是否创建虚拟环境? (y/n): " -n 1 -r
    echo
    if [[ $REPLY =~ ^[Yy]$ ]]; then
        python3 -m venv venv
        source venv/bin/activate
        echo "虚拟环境已创建并激活"
    fi
fi

# 检查依赖是否安装
echo "正在检查依赖包..."
python3 -c "import flask, sqlalchemy, pandas, openpyxl" 2>/dev/null
if [ $? -ne 0 ]; then
    echo "依赖包未安装，正在安装..."
    pip3 install -r requirements.txt
    if [ $? -ne 0 ]; then
        echo "依赖安装失败，请检查网络连接或手动安装"
        exit 1
    fi
fi

# 创建必要目录
mkdir -p logs uploads exports temp data

# 检查配置文件
if [ ! -f ".env" ]; then
    echo "配置文件不存在，正在创建默认配置..."
    cp env.example .env
    echo
    echo "请编辑 .env 文件配置数据库和AI模型信息"
    echo
fi

# 设置文件权限
chmod +x main.py

# 启动系统
echo "正在启动数据预处理系统..."
echo
python3 main.py

echo
echo "系统已停止运行"